package com.framework.core.text;

import lombok.extern.slf4j.Slf4j;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.math.BigInteger;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;
import java.nio.channels.FileChannel.MapMode;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;

@Slf4j
public class MD5Util {

    private static final String HEX_NUMS_STR = "0123456789ABCDEF";
    private static final Integer SALT_LENGTH = 12;

    /**
     * 将16进制字符串转换成字节数组
     *
     * @param hex
     * @return
     */
    public static byte[] hexStringToByte(String hex) {
        int len = (hex.length() / 2);
        byte[] result = new byte[len];
        char[] hexChars = hex.toCharArray();
        for (int i = 0; i < len; i++) {
            int pos = i * 2;
            result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4 | HEX_NUMS_STR
                    .indexOf(hexChars[pos + 1]));
        }
        return result;
    }

    /**
     * 验证口令是否合法
     *
     * @param password
     * @param passwordInDb
     * @return
     */
    public static boolean validPassword(String password, String passwordInDb){
        try {
            // 16进制字符串格式口令转换成字节数组
            byte[] pwdInDb = hexStringToByte(passwordInDb);
            // 声明盐变量
            byte[] salt = new byte[SALT_LENGTH];
            // 将盐从数据库中保存的口令字节数组中提取出�?
            System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);
            // 创建消息摘要对象
            MessageDigest md = MessageDigest.getInstance("MD5");
            // 将盐数据传入消息摘要对象
            md.update(salt);
            // 将口令的数据传给消息摘要对象
            md.update(password.getBytes("UTF-8"));
            // 生成输入口令的消息摘�?
            byte[] digest = md.digest();
            // 声明�?个保存数据库中口令消息摘要的变量
            byte[] digestInDb = new byte[pwdInDb.length - SALT_LENGTH];
            // 取得数据库中口令的消息摘�?
            System.arraycopy(pwdInDb, SALT_LENGTH, digestInDb, 0, digestInDb.length);
            // 比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同
            if (Arrays.equals(digest, digestInDb)) {
                // 口令正确返回口令匹配消息
                return true;
            } else {
                // 口令不正确返回口令不匹配消息
                return false;
            }
        } catch (NoSuchAlgorithmException | UnsupportedEncodingException e) {
            log.error(e.getMessage(),e);
        }
        return false;
    }

    /**
     * 加密
     * @param password 加密数据
     * @param key 加密向量
     * @return
     */
    public static String getEncryptedPwd(String password, String key){
        try {
            // 声明加密后的口令数组变量
            byte[] pwd = null;

            // 声明消息摘要对象
            MessageDigest md = null;
            // 创建消息摘要
            md = MessageDigest.getInstance("MD5");
            // 将盐数据传入消息摘要对象
            md.update(key.getBytes());
            // 将口令的数据传给消息摘要对象
            md.update(password.getBytes("UTF-8"));
            // 获得消息摘要的字节数�?
            byte[] digest = md.digest();

            // 将字节数组格式加密后的口令转化为16进制字符串格式的口令
            return AESUtil.parseByte2HexStr(digest);
        } catch (NoSuchAlgorithmException | UnsupportedEncodingException e) {
            log.error(e.getMessage(),e);
        }
        return null;
    }

    /**
     * 加密
     * @param srcStr
     * @return
     */
    public static String getMD5(String srcStr) {
        String s = null;
        char hexDigits[] = { // 用来将字节转换成 16 进制表示的字符
                '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
        if (srcStr == null) {
            return null;
        }
        try {
            byte[] source = srcStr.getBytes("UTF-8");
            MessageDigest md = MessageDigest.getInstance("MD5");
            md.update(source);
            byte tmp[] = md.digest(); // MD5 的计算结果是 128 位的长整数，
            char str[] = new char[16 * 2]; // 每个字节 16 进制表示的话，使用两个字符，
            int k = 0; // 表示转换结果中对应的字符位置
            for (int i = 0; i < 16; i++) { // 从第�?个字节开始，�? MD5 的每�?个字�?
                // 转换�? 16 进制字符的转�?
                byte byte0 = tmp[i]; // 取第 i 个字�?
                str[k++] = hexDigits[byte0 >>> 4 & 0xf]; // 取字节中�? 4 位的数字转换,
                // >>> 为�?�辑右移，将符号位一起右�?
                str[k++] = hexDigits[byte0 & 0xf]; // 取字节中�? 4 位的数字转换
            }
            s = new String(str); // 换后的结果转换为字符�?
        } catch (Exception e) {
            log.error("getMD5 error", e);
        }
        return s;
    }

    /**
     * 加密
     * @param date 加密数据
     * @param key 加密向量
     * @return
     */
    public static String getMD5(String date, String key){
        try {
            // 声明加密后的口令数组变量
            byte[] pwd = null;

            // 声明消息摘要对象
            MessageDigest md = null;
            // 创建消息摘要
            md = MessageDigest.getInstance("MD5");
            // 将盐数据传入消息摘要对象
            md.update(key.getBytes());
            // 将口令的数据传给消息摘要对象
            md.update(date.getBytes("UTF-8"));
            // 获得消息摘要的字节数�?
            byte[] digest = md.digest();

            // 将字节数组格式加密后的口令转化为16进制字符串格式的口令
            return AESUtil.parseByte2HexStr(digest);
        } catch (NoSuchAlgorithmException | UnsupportedEncodingException e) {
            log.error(e.getMessage(),e);
        }
        return null;
    }

    /**
     * 获取文件的MD5�?
     *
     * @param file
     * @return
     * @throws NoSuchAlgorithmException
     * @throws IOException
     */
    @SuppressWarnings("resource")
    public static String GetFileMD5(File file) throws IOException, NoSuchAlgorithmException {
        FileInputStream in = new FileInputStream(file);
        FileChannel fc = in.getChannel();
        MappedByteBuffer byteBuff = fc.map(MapMode.READ_ONLY, 0, fc.size()).load();
        MessageDigest dig = MessageDigest.getInstance("MD5");
        dig.update(byteBuff);
        BigInteger bigint = new BigInteger(1, dig.digest());
        return bigint.toString(16);
    }
}
